//
//  UIButton+CountDown.m
//  OilUnion
//
//  Created by itte on 2016/12/8.
//  Copyright © 2016年 funmain. All rights reserved.
//

#import "UIButton+CountDown.h"

@implementation UIButton (CountDown)
/**
 * 倒计时按钮
 * @param time     倒计时总时间
 * @param title    还没倒计时的title
 * @param subTitle 倒计时的title
 * @param mColor   还没倒计时的颜色
 * @param color    倒计时中的颜色
 */
-(void)startWithTime:(NSInteger)time
               title:(NSString *)title
            subTitle:(NSString *)subTitle
           mainColor:(UIColor *)mColor
          countColor:(UIColor *)color
{
    //倒计时时间
    __block NSInteger timeOut = time;
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    dispatch_source_t _timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, queue);
    //每秒执行一次
    dispatch_source_set_timer(_timer, dispatch_walltime(NULL, 0), 1.0 * NSEC_PER_SEC, 0);
    dispatch_source_set_event_handler(_timer, ^{
        //倒计时结束，关闭
        if (timeOut <= 0) {
            dispatch_source_cancel(_timer);
            dispatch_async(dispatch_get_main_queue(), ^{
                self.backgroundColor = mColor;
                [self setTitle:title forState:UIControlStateNormal];
                self.userInteractionEnabled = YES;
            });
        } else {
            int allTime = (int)time + 1;
            int seconds = timeOut % allTime;
            NSString *timeStr = [NSString stringWithFormat:@"%0d", seconds];
            dispatch_async(dispatch_get_main_queue(), ^{
                self.backgroundColor = color;
                [self setTitle:[NSString stringWithFormat:@"%@%@",timeStr,subTitle] forState:UIControlStateNormal];
                self.userInteractionEnabled = NO;
            });
            timeOut--;
        }
    });
    dispatch_resume(_timer);
}
@end
